"""E2E proof for #797: exclude_dynamic_sections cross-user caching."""

import anyio

from claude_agent_sdk import (
    AssistantMessage,
    ClaudeAgentOptions,
    ResultMessage,
    TextBlock,
    query,
)

CLI = "/Users/qing/code/wt-cli-e2e/build-external/cli.js"
ALICE = "/Users/qing/code/tmp/e2e-alice-git"
BOB = "/Users/qing/code/tmp/e2e-bob-git"
PROMPT = (
    "What is my primary working directory? Reply with only the absolute path, "
    "nothing else."
)


async def run(label: str, cwd: str, exclude: bool) -> dict:
    sp: dict = {"type": "preset", "preset": "claude_code"}
    if exclude:
        sp["exclude_dynamic_sections"] = True
    opts = ClaudeAgentOptions(
        cwd=cwd,
        max_turns=1,
        model="claude-haiku-4-5-20251001",
        system_prompt=sp,
        allowed_tools=[],
        cli_path=CLI,
    )
    answer = ""
    usage: dict = {}
    async for msg in query(prompt=PROMPT, options=opts):
        if isinstance(msg, AssistantMessage):
            for block in msg.content:
                if isinstance(block, TextBlock):
                    answer += block.text
        elif isinstance(msg, ResultMessage):
            usage = msg.usage or {}
    cc = usage.get("cache_creation_input_tokens", "-")
    cr = usage.get("cache_read_input_tokens", "-")
    print(
        f"{label} | cwd={cwd} | answer={answer.strip()!r} | "
        f"cache_creation={cc} cache_read={cr}"
    )
    return {"answer": answer.strip(), "cc": cc, "cr": cr}


async def main() -> None:
    print("=== PR #797 e2e: exclude_dynamic_sections (Python SDK) ===\n")
    r1 = await run("1-default-alice ", ALICE, exclude=False)
    r2 = await run("2-default-bob   ", BOB, exclude=False)
    r3 = await run("3-exclude-alice ", ALICE, exclude=True)
    r4 = await run("4-exclude-bob   ", BOB, exclude=True)

    print("\n=== Analysis ===")
    ok1 = "e2e-alice-git" in r1["answer"]
    ok3 = "e2e-alice-git" in r3["answer"] and "e2e-bob-git" in r4["answer"]
    print(f"default mode knows cwd (no regression):     {'PASS' if ok1 else 'FAIL'}")
    print(f"exclude mode still knows cwd (re-inject):   {'PASS' if ok3 else 'FAIL'}")
    print(f"default cross-cwd cache_creation:           {r2['cc']}")
    print(f"exclude cross-cwd cache_creation:           {r4['cc']}")
    if isinstance(r2["cc"], int) and isinstance(r4["cc"], int):
        delta = r2["cc"] - r4["cc"]
        print(f"savings (lower is better in exclude):       {delta} tokens")


anyio.run(main)

"""E2E: exclude_dynamic_sections with ~6k-token append (PR #797)."""

import asyncio

from claude_agent_sdk import ClaudeAgentOptions, query

CLI = "<path-to-unreleased-cli>"
ALICE = "/Users/qing/code/tmp/e2e-alice-git"
BOB = "/Users/qing/code/tmp/e2e-bob-git"
PROMPT = (
    "What is my primary working directory? Reply with only the absolute path, "
    "nothing else."
)
# ~74k chars ≈ ~6k tokens (highly repetitive text tokenizes efficiently)
APPEND = ("You are a helpful coding assistant. " * 10 + "\n") * 200


async def run(label: str, cwd: str, exclude: bool) -> dict:
    sp: dict = {"type": "preset", "preset": "claude_code", "append": APPEND}
    if exclude:
        sp["exclude_dynamic_sections"] = True
    opts = ClaudeAgentOptions(
        cwd=cwd,
        max_turns=1,
        model="claude-haiku-4-5-20251001",
        system_prompt=sp,
        allowed_tools=[],
        cli_path=CLI,
    )
    answer = ""
    usage: dict = {}
    err = None
    try:
        async for msg in query(prompt=PROMPT, options=opts):
            name = type(msg).__name__
            if name == "AssistantMessage":
                for block in getattr(msg, "content", []):
                    if type(block).__name__ == "TextBlock":
                        answer += getattr(block, "text", "")
            elif name == "ResultMessage":
                usage = getattr(msg, "usage", None) or {}
    except Exception as e:  # noqa: BLE001
        err = f"{type(e).__name__}: {e}"
    cc = usage.get("cache_creation_input_tokens", "-")
    cr = usage.get("cache_read_input_tokens", "-")
    line = (
        f"{label:16s} | cwd={cwd} | answer={answer.strip()!r} | "
        f"cache_creation={cc} cache_read={cr}"
    )
    if err:
        line += f" | ERROR={err}"
    print(line)
    return {"answer": answer.strip(), "cc": cc, "cr": cr, "err": err}


async def main() -> None:
    print("=== PR 797 e2e: exclude_dynamic_sections with ~6k-token append ===")
    print(f"CLI: {CLI}")
    print(f"append size: {len(APPEND)} chars\n")
    r1 = await run("1-default-alice", ALICE, exclude=False)
    r2 = await run("2-default-bob", BOB, exclude=False)
    r3 = await run("3-exclude-alice", ALICE, exclude=True)
    r4 = await run("4-exclude-bob", BOB, exclude=True)

    print("\n=== Analysis ===")
    ok1 = "e2e-alice-git" in r1["answer"]
    ok34 = "e2e-alice-git" in r3["answer"] and "e2e-bob-git" in r4["answer"]
    print(f"default mode knows cwd (no regression):     {'PASS' if ok1 else 'FAIL'}")
    print(f"exclude mode still knows cwd (re-inject):   {'PASS' if ok34 else 'FAIL'}")
    d = r2["cc"] if isinstance(r2["cc"], int) else 0
    e = r4["cc"] if isinstance(r4["cc"], int) else 0
    if d and e:
        pct = round(100 * (d - e) / d)
        print(f"cross-cwd cache_creation: {d} -> {e} (-{pct}%, {d - e} tokens saved)")
    else:
        print(f"cross-cwd cache_creation: {d} -> {e}")


if __name__ == "__main__":
    asyncio.run(main())